package com.lhj.xxsc.cart.interceptor;

/**
 * @author 罗惠江
 * @version 1.0
 * @email 1511619294@qq.com
 * @date 2020/12/30 15:17
 */

import com.xxsc.common.constant.AuthServerConstant;
import com.xxsc.common.constant.CartConstant;
import com.xxsc.common.vo.MemberRespVo;
import com.lhj.xxsc.cart.to.UserInfoTo;
import org.springframework.stereotype.Component;
import org.springframework.util.StringUtils;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.Cookie;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.util.UUID;

/**
 * 执行目标方法之前，判断用户登录状态
 *
 * @author son
 */
@Component
public class CartInterceptor implements HandlerInterceptor {

	public static ThreadLocal<UserInfoTo> threadLocal = new ThreadLocal<>();

	@Override
	public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
		UserInfoTo userInfoTo = new UserInfoTo();

		HttpSession session = request.getSession();
		MemberRespVo memberRespVo = (MemberRespVo) session.getAttribute(AuthServerConstant.LOGIN_USER);
		if (memberRespVo != null) {
			userInfoTo.setUserId(memberRespVo.getId());
		}

		Cookie[] cookies = request.getCookies();
		if (cookies != null && cookies.length > 0) {
			for (Cookie cookie : cookies) {
				String name = cookie.getName();
				if (CartConstant.TEMP_USER_COOKIE_NAME.equals(name)) {
					userInfoTo.setUserKey("" + cookie.getValue());
					userInfoTo.setTempUser(true);
				}
			}
		}
		// 没有临时id 创建临时id
		if (StringUtils.isEmpty(userInfoTo.getUserKey())) {
			String uuid = UUID.randomUUID().toString();
			userInfoTo.setUserKey(uuid);
		}
		// 目标方法执行之前
		threadLocal.set(userInfoTo);
		return true;
	}

	/**
	 * 业务执行之后;分配临时用户，让浏览器保存
	 *
	 * @param request
	 * @param response
	 * @param handler
	 * @param modelAndView
	 * @throws Exception
	 */
	@Override
	public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {

		UserInfoTo userInfoTo = threadLocal.get();

		// 没有临时用户一定保存一个临时用户
		if (!userInfoTo.isTempUser()) {
			Cookie cookie = new Cookie(CartConstant.TEMP_USER_COOKIE_NAME, userInfoTo.getUserKey());
			cookie.setDomain("xxsc.com");
			cookie.setMaxAge(CartConstant.TEMP_USER_COOKIE_TIME);
			response.addCookie(cookie);
		}
	}
}
